package com.example.actionsignin.utils;

import android.app.Activity;
import android.content.Context;
import android.content.SharedPreferences;
import android.os.Bundle;
import android.os.Environment;
import android.util.Log;

import com.example.actionsignin.entity.VoiceEntity;
import com.google.gson.Gson;
import com.iflytek.cloud.ErrorCode;
import com.iflytek.cloud.InitListener;
import com.iflytek.cloud.RecognizerListener;
import com.iflytek.cloud.RecognizerResult;
import com.iflytek.cloud.SpeechConstant;
import com.iflytek.cloud.SpeechError;
import com.iflytek.cloud.SpeechRecognizer;
import com.iflytek.cloud.ui.RecognizerDialog;
import com.iflytek.cloud.ui.RecognizerDialogListener;

import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.List;


/**
 * 语音识别
 * Created by 旭 on 2017/11/6/006.
 */

public class VoiceUtil {

    private static final String TAG = "VoiceUtil";
    public static String PREFER_NAME = "com.voice.recognition";
    Context mContext;

    OnVoiceListener mOnVoiceListener;

    SpeechRecognizer mIat;
    RecognizerDialog mIatDialog;
    SharedPreferences mSharedPreferences;

    HashMap<String, String> mIatResults = new LinkedHashMap<>();

    int ret = 0;
    String mEngineType = SpeechConstant.TYPE_CLOUD;

    StringBuffer mStringBuffer = new StringBuffer();

    InitListener mInitListener = new InitListener() {
        @Override
        public void onInit(int i) {
//            Log.d(TAG, "onInit: " + i);
            if (i != ErrorCode.SUCCESS) {
                Log.d(TAG, "onInit: 初始化错误，错误码：" + i);
            }
        }
    };

    public VoiceUtil(Context context)
    {
        this.mContext = context;
        initData();
    }

    private void initData() {

        // 初始化识别无UI识别对象
        // 使用SpeechRecognizer对象，可根据回调消息自定义界面；
        mIat = SpeechRecognizer.createRecognizer(mContext, mInitListener);
        // 初始化听写Dialog，如果只使用有UI听写功能，无需创建SpeechRecognizer
        // 使用UI听写功能，请根据sdk文件目录下的notice.txt,放置布局文件和图片资源
        mIatDialog = new RecognizerDialog(mContext, mInitListener);
        mSharedPreferences = mContext.getSharedPreferences(VoiceUtil.PREFER_NAME, Activity.MODE_PRIVATE);
    }

    private void checkSoIsInstallSucceed() {
        if (null == mIat) {
            // 创建单例失败，与 21001 错误为同样原因，参考 http://bbs.xfyun.cn/forum.php?mod=viewthread&tid=9688
            Log.d(TAG, "checkSoIsInstallSucceed: 创建对象失败，请确认 libmsc.so 放置正确，且有调用 createUtility 进行初始化");
        }
    }

    private void setParam() {
        // 清空参数
        mIat.setParameter(SpeechConstant.PARAMS, null);

        // 设置听写引擎
        mIat.setParameter(SpeechConstant.ENGINE_TYPE, mEngineType);
        // 设置返回结果格式
        mIat.setParameter(SpeechConstant.RESULT_TYPE, "json");

        String lag = mSharedPreferences.getString("iat_language_preference",
                "mandarin");
        if (lag.equals("en_us")) {
            // 设置语言
            mIat.setParameter(SpeechConstant.LANGUAGE, "en_us");
        } else {
            // 设置语言
            mIat.setParameter(SpeechConstant.LANGUAGE, "zh_cn");
            // 设置语言区域
            mIat.setParameter(SpeechConstant.ACCENT, lag);
        }

        // 设置语音前端点:静音超时时间，即用户多长时间不说话则当做超时处理
        mIat.setParameter(SpeechConstant.VAD_BOS, mSharedPreferences.getString("iat_vadbos_preference", "4000"));

        // 设置语音后端点:后端点静音检测时间，即用户停止说话多长时间内即认为不再输入， 自动停止录音
        mIat.setParameter(SpeechConstant.VAD_EOS, mSharedPreferences.getString("iat_vadeos_preference", "1000"));

        // 设置标点符号,设置为"0"返回结果无标点,设置为"1"返回结果有标点
        mIat.setParameter(SpeechConstant.ASR_PTT, mSharedPreferences.getString("iat_punc_preference", "1"));

        // 设置音频保存路径，保存音频格式支持pcm、wav，设置路径为sd卡请注意WRITE_EXTERNAL_STORAGE权限
        // 注：AUDIO_FORMAT参数语记需要更新版本才能生效
        mIat.setParameter(SpeechConstant.AUDIO_FORMAT, "wav");
        mIat.setParameter(SpeechConstant.ASR_AUDIO_PATH, Environment.getExternalStorageDirectory() + "/msc/iat.wav");
    }

    /**
     * 听写UI监听器
     */
    private RecognizerDialogListener mRecognizerDialogListener = new RecognizerDialogListener() {
        public void onResult(RecognizerResult results, boolean isLast) {
            printResult(results);
        }

        /**
         * 识别回调错误.
         */
        public void onError(SpeechError error) {
            showTip(error.getPlainDescription(true));
        }

    };

    /**
     * 听写监听器。
     */
    private RecognizerListener mRecognizerListener = new RecognizerListener() {

        @Override
        public void onBeginOfSpeech() {
            // 此回调表示：sdk内部录音机已经准备好了，用户可以开始语音输入
            showTip("开始说话");
        }

        @Override
        public void onError(SpeechError error) {
            // Tips：
            // 错误码：10118(您没有说话)，可能是录音机权限被禁，需要提示用户打开应用的录音权限。
            // 如果使用本地功能（语记）需要提示用户开启语记的录音权限。
            showTip(error.getPlainDescription(true));
        }

        @Override
        public void onEndOfSpeech() {
            // 此回调表示：检测到了语音的尾端点，已经进入识别过程，不再接受语音输入
            showTip("结束说话");
        }

        @Override
        public void onResult(RecognizerResult results, boolean isLast) {
            Log.d(TAG, "onResult: " + results.getResultString());
            printResult(results);

            if (isLast) {
                // TODO 最后的结果
                isTrue(mStringBuffer.toString());
            }
        }

        @Override
        public void onVolumeChanged(int volume, byte[] data) {
            showTip("当前正在说话，音量大小：" + volume);
//            Log.d(TAG, "返回音频数据: "+data.length);
        }

        @Override
        public void onEvent(int eventType, int arg1, int arg2, Bundle obj) {
            // 以下代码用于获取与云端的会话id，当业务出错时将会话id提供给技术支持人员，可用于查询会话日志，定位出错原因
            // 若使用本地能力，会话id为null
            //  if (SpeechEvent.EVENT_SESSION_ID == eventType) {
            //      String sid = obj.getString(SpeechEvent.KEY_EVENT_SESSION_ID);
            //      Log.d(TAG, "session id =" + sid);
            //  }
        }
    };

    private void printResult(RecognizerResult results) {
        Gson gson = new Gson();
        String text = results.getResultString();
        VoiceEntity resultEntity = new VoiceEntity();

        resultEntity = gson.fromJson(text, VoiceEntity.class);

        List<VoiceEntity.Voice> voiceEntities = resultEntity.getWs();
        for (int i = 0; i < voiceEntities.size(); i++) {
            List<VoiceEntity.Voice.WordEntity> wordEntities = voiceEntities.get(i).getCw();
            for (int j = 0; j < wordEntities.size(); j++) {
                String str = wordEntities.get(j).getW();
                mStringBuffer.append(str);
            }
        }

        Log.d(TAG, "printResult: " + mStringBuffer.toString());
        isTrue(mStringBuffer.toString());
    }

    private void showTip(final String str) {
        ToastUtil.INSTANCE(mContext,str);
    }

    public void startVoice()
    {
        checkSoIsInstallSucceed();
        mStringBuffer.delete(0, mStringBuffer.length());
        mIatResults.clear();
        setParam();
        boolean isShowDialog = true;
        if (isShowDialog) {
            //显示听写对话框
            mIatDialog.setListener(mRecognizerDialogListener);
            mIatDialog.show();
        } else {
            //不显示听写对话框
            ret = mIat.startListening(mRecognizerListener);
            if (ret != ErrorCode.SUCCESS) {
                Log.d(TAG, "onClick: 失败");
            } else {
                Log.d(TAG, "onClick: 成功");
            }
        }
    }

    public void stopVoice()
    {
        checkSoIsInstallSucceed();
        mStringBuffer.delete(0,mStringBuffer.length());
        if (mIat != null)
        {
            mIat.cancel();
            mIat.destroy();
        }
    }

    private void isTrue(String string)
    {
        if (string == null || string.equals("")){
            return;
        }
        Log.d(TAG, "isTrue: " + string);
        if (string.equals("签到。"))
        {
            Log.d(TAG, "isTrue: 111");
            ToastUtil.INSTANCE(mContext,"识别成功!");
            mOnVoiceListener.onVoice();
        }else {
            ToastUtil.INSTANCE(mContext,"识别失败，请重新识别!");
        }
    }

    public void setVoiceListener(OnVoiceListener onVoiceListener)
    {
        this.mOnVoiceListener = onVoiceListener;
    }

    public interface OnVoiceListener
    {
        void onVoice();
    }

}
